package exercise01;

import java.util.*;

// 全排列
// 无重复数
public class Day230703 {
    List<List<Integer>> result = new ArrayList<>();
    Deque<Integer> sub = new ArrayDeque<>();

    public List<List<Integer>> permute(int[] nums) {
        backtrack(nums);
        return result;
    }

    public void backtrack(int[] nums) {
        if (sub.size() == nums.length) {
            result.add(new ArrayList(sub));
            return;
        }
        for (int num : nums) {
            if (!sub.contains(num)) {
                sub.add(num);
                backtrack(nums);
                sub.removeLast();
            }
        }
    }
}

// 全排列 II
// 有重复数
class Solution {
    List<List<Integer>> result = new ArrayList<>();
    Deque<Integer> sub = new ArrayDeque<>();
    boolean[] visited;      // 标记已选择的数

    public List<List<Integer>> permuteUnique(int[] nums) {
        visited = new boolean[nums.length];
        Arrays.sort(nums);  // 排序
        backtrack(nums);
        return result;
    }

    public void backtrack(int[] nums) {
        if (sub.size() == nums.length) {
            result.add(new ArrayList(sub));
            return;
        }
        for (int i = 0; i < nums.length; i++) {
            if (visited[i] || (i > 0 && nums[i] == nums[i - 1] && !visited[i- 1])) {
                // 已选择的数, 不能再选
                // 未选择的数, 但是与上一个数相同, 上一个数未选择, 这个也不能选
                continue;
            }
            sub.add(nums[i]);
            visited[i] = true;  // 标记
            backtrack(nums);
            sub.removeLast();
            visited[i] = false; // 去除标记
        }
    }
}
